package leetcode.leetcode103;

import java.util.*;

/**
 * @author czt
 * @version 1.0
 * @since 2024/12/23
 *
 * 使用双端队列实现Z遍历二叉树
 * 和层序遍历二叉树相同，只不过需判断奇偶层，如果是偶数层则在头部插入，形成逆序
 */
public class Leetcode103 {

    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<>();
        if (root == null) {
            return result;
        }

        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);

        boolean isOdd = true; // 判断是否是奇偶数

        while (!queue.isEmpty()) {
            int levelSize = queue.size();
            LinkedList<Integer> currentLevel = new LinkedList<>();

            for (int i = 0; i < levelSize; i++) {
                TreeNode node = queue.poll(); // 从队列中获取节点
                if (isOdd) {
                    // 第一次为基数，则尾部添加
                    currentLevel.offerLast(node.val); // 添加节点值到当前层
                }else {
                    // 为偶数时，头部添加
                    currentLevel.offerFirst(node.val);
                }

                // 将左右子节点加入队列
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            // 等遍历下一层的时候，奇偶取一下反
            isOdd = !isOdd;
            result.add(currentLevel); // 将当前层的结果加入最终结果
        }

        return result;
    }
}
